package util;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.agenda.entity.Atividade;
  
public class AtividadeDataModel extends ListDataModel implements SelectableDataModel<Atividade> {    
  
    public AtividadeDataModel() {  
    }  
  
    public AtividadeDataModel(List<Atividade> data) {  
        super(data);  
    }  
      
    @Override  
    public Atividade getRowData(String rowKey) {  
        //In a real app, a more efficient way like a query by rowKey should be implemented to deal with huge data  
          
        List<Atividade> atividades = (List<Atividade>) getWrappedData();  
          
        for(Atividade atividade : atividades) {  
            if(atividade.getIdAtividade().equals(rowKey))  
                return atividade;  
        }  
          
        return null;  
    }  

	@Override  
    public Object getRowKey(Atividade atividade) {  
        return atividade.getIdAtividade();  
    }  
}  